# !/usr/bin/env python3
# -*- coding: utf-8 -*-


class Intent:
    """
    意图类: 用于记录用户的意图
    """

    intent_keyword = {
        'INFORMATION': [['介绍'], ['是什么']],
        'TIME': [
            ['几点'], ['时间'], ['时候'], ['今天'], ['明天'], ['后天'], ['周一'],
            ['星期一'], ['周二'], ['星期二'], ['周三'], ['星期三'], ['星期四'], ['周四'],
            ['星期五'], ['周五'], ['星期六'], ['周六'], ['星期日'], ['周日'], ['礼拜天'],
            ['哪天'], ['周几'], ['星期几'], ['哪周'], ['现在'], ['开放'], ['上午'],
            ['下午'], ['晚上']
        ],
        'POSITION': ['去哪', '哪里', '在哪', '地方', '位置'],
        'CONTAIN': ['包括', '包含', '有什么', '有哪些', '多少', '有几', '含'],
        'BUSINESS': ['怎么', '如何'],
        'MULTIPLE': ['办理', '推荐'],
        'CONDITION': [['借', '书', '证'], ['借阅', '证']]
    }

    def __init__(self):
        self.intent = 'normal'
        self.mul_intent = ['recommend']

    def reset_intent(self, intent):
        self.intent = intent

    def set_intent(self, respond):
        if respond.find("_") != -1:
            intent = respond.split("_")[-1]
            self.intent = intent
        else:
            self.intent = 'normal'

    def get_intent(self):
        return self.intent
